package TreeSetDemo;

import java.util.Comparator;
import java.util.TreeSet;

public class TreeSetDemo1 {
//    需求:请自行选择比较器排序和自然排序两种方式;
//    要求:存入四个字符串,"c","ab","df","qwer"
//    按照长度排序,如果一样长则按照首字母排序

    public static void main(String[] args) {
        //1.创建集合
        //o1：表示当前要添加的元素
        //o2:表示已经在红黑树存在的元素
        TreeSet<String> ts = new TreeSet<>((o1,  o2)-> {
                //按照长度排序
                int i = o1.length() - o2.length();
                int i1 = i == 0 ? o1.compareTo(o2) : i;
                return i1;
            }
        );

        //添加元素
        ts.add("c");
        ts.add("ab");
        ts.add("cd");
        ts.add("ef");

        System.out.println(ts);
    }
}
